package cn.itsource.system.service.impl;



import cn.itsource.basic.PageList;
import cn.itsource.basic.base.query.BaseQuery;
import cn.itsource.system.domain.Menu;
import cn.itsource.system.mapper.MenuMapper;
import cn.itsource.system.service.IMenuService;
import cn.itsource.basic.base.service.impl.BaseServiceImpl;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

/**
 * <p>
 *  服务实现类
 * </p>
 *
 * @author zhi
 * @since 2022-07-14
 */
@Service
public class MenuServiceImpl extends BaseServiceImpl<Menu> implements IMenuService {
    @Autowired
    private MenuMapper menuMapper;


    @Override
    public List<Menu> queryTree() {
        //查询出所有数据
        List<Menu> allDepts = menuMapper.loadAll();

        // 把所有的数据转换成Map结构--后面查找自己的上级使用
        HashMap<Long, Menu> menuMap = new HashMap<>();
        for(Menu menu : allDepts){
            menuMap.put(menu.getId(), menu);
        }
        //准备要返回的无限极树，把顶级部门全放到fistDepts集合中
        List<Menu> firstDepts = new ArrayList<>();

        for (Menu menu : allDepts) {
            //1.从所有数据中找出第一层级 firstDept   我们需要返回的数据
            if(menu.getParentId() == null || menu.getParentId().intValue() == 0 ){
                firstDepts.add(menu);//最顶层的数据
            }else{
                //这里拿到的是有爹的部门
                Menu dadMenu = menuMap.get(menu.getParentId());
                //判断爹时候存在
                if(dadMenu != null){
                    //如果存在，把自己装到爹的children里
                    dadMenu.getChildren().add(menu);
                }
            }
        }
        return firstDepts;
    }



    @Override
    @Transactional
    public void add(Menu menu) {
        menuMapper.insert(menu);
        this.update(menu);
    }

    @Override
    public PageList<Menu> queryPage(BaseQuery query) {
        Long count = menuMapper.count(query);
        List<Menu> menus = menuMapper.pageData(query);
        for (Menu menu : menus) {
            Menu menu1 = menuMapper.loadById(menu.getParentId());
            menu.setParent(menu1);
        }
        return new PageList<>(count,menus);
    }


}
